#pragma once

#include <iostream>
#include <sys/types.h>
#include <unistd.h>
#include <signal.h>
#include <string>
#include <sys/stat.h>
#include <fcntl.h>

const static std::string root = "/";
const static std::string dev_null = "/dev/null";

void Daemon(bool ischadir, bool isclose)
{
    // 1. 忽略不必要的信号
    signal(SIGPIPE, SIG_IGN);
    signal(SIGCHLD, SIG_IGN);

    // 2. 关闭父进程
    if (fork() > 0)
        exit(0);

    // 3. 创建新的会话
    setsid();

    // 4. 修改文件路径
    if (ischadir)
        chdir(root.c_str());

    // 5. 重定向文件描述符
    if (isclose)
    {
        ::close(0);
        ::close(1);
        ::close(2);
    }
    else
    {
        int fd = open(dev_null.c_str(), O_RDWR);
        if (fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            ::close(fd);
        }
    }
}
